package com.dkm.nettysocketdemo.myWebsite.action;

import com.dkm.nettysocketdemo.myWebsite.domain.HelmetImg;
import com.dkm.nettysocketdemo.myWebsite.service.HelmetImgService;
import com.dkm.nettysocketdemo.untils.Message;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.util.List;

/**
 * @ClassName : HelmetImgAction  //类名
 * @Description : 安全帽图片Action  //描述
 * @Author : 易嵘  //作者
 * @Date: 2021-03-23 09:53  //时间
 */
@RestController
@RequestMapping("HelmetImgAction")
public class HelmetImgAction {

    @Autowired
    HelmetImgService imgService;

    /**
     * 新增安全帽图片
     *
     * @param helmetId    安全帽id
     * @param file 图片文件
     * @return 回调消息
     */
    @PostMapping("addImage")
    public Message addImage(@RequestParam("helmetId") String helmetId, @RequestBody MultipartFile file) {
        Message message = new Message();
        int row = imgService.addImage(helmetId, file);
        message.setCode(200);
        message.setMsg("新增成功");
        message.setData(row);
        return message;
    }

    /**
     * 根据安全帽id查询所有图片
     *
     * @param helmetId 安全帽id
     * @return 安全帽图片集合
     */
    @GetMapping("findHelmetIdAll/{helmetId}")
    public Message findHelmetIdAll(@PathVariable(value = "helmetId") String helmetId) {
        Message message = new Message();
        List<HelmetImg> imgs = imgService.findHelmetIdAll(helmetId);
        message.setData(imgs);
        message.setMsg("查询成功");
        message.setCode(200);
        return message;
    }

}
